import pickle

import torch

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

ws = pickle.load(open("imdb/model/ws.pkl", "rb"))
length = 200  # 每个句子长度

batch_size = 128
test_batch_size = 128

embedding_dim = 100

hidden_size = 128  # 隐藏层中lstm的个数
num_layers = 2  # 隐藏层的个数
dropout = 0.5
bidirectional = True
